export IMAGENET_TRAIN_DATASET=path_of_dataset
export TORCH_HOME=$PWD/data
export IMAGENET_INFER_CHECKPOINT=path_of_infer_checkpoint
